# 使用ROCm官方镜像作为基础镜像
FROM rocm/pytorch:rocm5.4.2_ubuntu20.04_py3.8_pytorch_2.0.1

# 设置工作目录
WORKDIR /app

# 复制项目文件
COPY . /app/

# 安装依赖
RUN pip install --no-cache-dir -r requirements.txt

# 设置环境变量
ENV API_HOST=0.0.0.0
ENV API_PORT=8000
ENV API_DEBUG=False
ENV MODEL_DEVICE=cuda
ENV MODEL_PRECISION=fp16
ENV ROCM_VISIBLE_DEVICES=0

# 暴露端口
EXPOSE 8000

# 启动命令
CMD ["python", "main.py"]